package com.huashi.utils;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;

import javax.annotation.Resource;
import java.util.Collection;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.TimeUnit;


@Service
public class RedisUtil {

    private static final Logger log = LoggerFactory.getLogger(RedisUtil.class);

    @Resource
    RedisTemplate<String,Object> redisTemplate;

    public void clean(){
        Set<String> keys = redisTemplate.keys("*");
        redisTemplate.delete(keys);
    }


    /**
     * 获取所有key
     * @param keys
     * @return
     */
    public Set<String> keys(String keys){
        try{
            return redisTemplate.keys(keys);
        }catch (Exception e){
            log.error("获取所有key失败",e);
            return null;
        }
    }


    /**
     * 指定缓存失效时间
     * @param key
     * @param time
     * @param timeUnit
     */
    public void expire(String key, Long time, TimeUnit timeUnit){
        try {
            if (time>0){
                redisTemplate.expire(key,time,timeUnit);
            }
        }catch (Exception e){
            log.error("指定缓存失效时间失败");
        }
    }

    /**
     * 根据key获取过期时间
     * @param key
     * @param timeUnit
     * @return
     */
    public long getExpire(String key,TimeUnit timeUnit){
        return redisTemplate.getExpire(key,timeUnit);
    }

    /**
     * 判断key是否存在
     * @param key
     * @return
     */
    public boolean hasKey(String key){
        try{
            return redisTemplate.hasKey(key);
        }catch (Exception e){
            log.error("判断key是否存在失败",e);
            return false;
        }
    }


    /**
     * 删除缓存
     * @param key
     */
    @SuppressWarnings("unchecked")
    public void del(String... key){
        if (key.length == 1){
            redisTemplate.delete(key[0]);
        }else {
            redisTemplate.delete((Collection<String>) CollectionUtils.arrayToList(key));
        }
    }

    /**
     * 普通缓存获取
     * @param key
     * @return
     */
    public Object getKey(String key){
        return key == null ? null : redisTemplate.opsForValue().get(key);
    }

    /**
     * 普通缓存放入
     * @param key
     * @param value
     */
    public void setKey(String key,Object value){
        try {
            redisTemplate.opsForValue().set(key,value);
        }catch (Exception e){
            log.error("普通缓存放入失败",e);
        }
    }


    /**
     * 普通缓存放入并设置时间
     * @param key
     * @param value
     * @param time
     */
    public void setKey(String key, Object value, Long time){
        try{
            if (time >0){
                redisTemplate.opsForValue().set(key,value,time,TimeUnit.SECONDS);
            }else {
                setKey(key,value);
            }
        }catch (Exception e){
            log.error("普通缓存放入并设置时间失败",e);
        }
    }

    /**
     * 普通缓存放入并设置时间
     * @param key
     * @param value
     * @param time
     * @param timeUnit
     */
    public void setKey(String key, Object value, Long time, TimeUnit timeUnit){
        try{
            if (time >0){
                redisTemplate.opsForValue().set(key,value,time,timeUnit);
            }else {
                setKey(key,value);
            }
        }catch (Exception e){
            log.error("普通缓存放入并设置时间失败",e);
        }
    }


    /**
     * 递增
     * @param key
     * @param delta
     * @return
     */
    public long incr(String key, long delta){
        if (delta<0){
            throw new RuntimeException("递增因子必须大于0");
        }
        return redisTemplate.opsForValue().increment(key, delta);
    }

    /**
     * 递减
     * @param key
     * @param delta
     * @return
     */
    public long decr(String key,long delta){
        if (delta<0){
            throw new RuntimeException("递减因子必须大于零");
        }
        return redisTemplate.opsForValue().increment(key,delta);
    }


    /**
     * HashGet
     * @param key
     * @param hashKey
     * @return
     */
    public Object hget(String key,  Object hashKey){
        return redisTemplate.opsForHash().get(key,hashKey);
    }

    /**
     * 获取hashKey对应的所有键值
     * @param key
     * @return
     */
    public Map<Object,Object> hmget(String key){
        return redisTemplate.opsForHash().entries(key);
    }

    /**
     * 获取hashKey对应的所有键值
     * @param key
     * @param hashKey
     * @return
     */
    public Object hmget(String key, Object hashKey){
        return redisTemplate.opsForHash().get(key, hashKey);
    }

    /**
     * 获取hashKey对应的所有键值
     * @param key
     * @param hashKeys
     * @return
     */
    public List<Object> hmgetLsit(String key, Collection<Object> hashKeys){
        return redisTemplate.opsForHash().multiGet(key, hashKeys);
    }

    /**
     * 获取HashSet对应的所有键值
     * @param key
     * @param map
     * @return
     */
    public boolean hmset(String key,Map<Object,Object> map){
        try {
            redisTemplate.opsForHash().putAll(key,map);
            return true;
        }catch (Exception e){
            log.error("HashSet缓存失败",e);
            return false;
        }
    }

    /**
     * HashSet 并设置时间
     * @param key
     * @param map
     * @param time
     * @param timeUnit
     * @return
     */
    public boolean hmset(String key, Map<Object,Object> map,long time,TimeUnit timeUnit){
        try{
            redisTemplate.opsForHash().putAll(key,map);
            if (time > 0){
                expire(key,time,timeUnit);
            }
            return true;
        }catch (Exception e){
            log.error("HashSet缓存并设置时间失败",e);
            return false;
        }
    }

    /**
     * 向一张hash表中放入数据,如果不存在将创建失败
     * @param key
     * @param hashKey
     * @param value
     * @return
     */
    public boolean hset(String key,Object hashKey, Object value){
        try{
            redisTemplate.opsForHash().put(key, hashKey, value);
            return true;
        }catch (Exception e){
            log.error("向一张hash表中放入数据,如果不存在将创建失败",e);
            return false;
        }
    }

    /**
     * 向一张hash表中放入数据,如果不存在将创建
     * @param key
     * @param hashKey
     * @param value
     * @param time
     * @param timeUnit
     * @return
     */
    public boolean hset(String key, String hashKey, Object value,long time ,TimeUnit timeUnit){
        try{
            redisTemplate.opsForHash().put(key,hashKey,value);
            if (time > 0){
                expire(key,time,timeUnit);
            }
            return true;
        }catch (Exception e){
            log.error("向一张hash表中放入数据,如果不存在将创建失败",e);
            return false;
        }
    }

    /**
     * 删除hash表中的值
     * @param key
     * @param hashKey
     */
    public void hdel(String key, Object... hashKey){
        redisTemplate.opsForHash().delete(key,hashKey);
    }


    /**
     * 判断hash表中是否有该项的值
     * @param key
     * @param hashKey
     * @return
     */
    public boolean hHashKey(String key, Object hashKey){
        return redisTemplate.opsForHash().hasKey(key, hashKey);
    }
}
